// Copyright (c) Huawei Technologies Co., Ltd. 2025. All rights reserved

#include "tpp/alphafold/bind.h"
#include "utils/memory.h"
#include <torch/extension.h>

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m)
{
    alphafold::bind(m);
    m.def("device", &kpex::device);
}